#include "CvvImage.h"
//////////////////////////////////////////////////////////////////////
// Construction/Destruction
//////////////////////////////////////////////////////////////////////
CV_INLINE RECT NormalizeRect(RECT r);
CV_INLINE RECT NormalizeRect(RECT r)
{
    int t;
    if (r.left > r.right)
    {
        t = r.left;
        r.left = r.right;
        r.right = t;
    }
    if (r.top > r.bottom)
    {
        t = r.top;
        r.top = r.bottom;
        r.bottom = t;
    }

    return r;
}
CV_INLINE CvRect RectToCvRect(RECT sr);
CV_INLINE CvRect RectToCvRect(RECT sr)
{
    sr = NormalizeRect(sr);
    return cvRect(sr.left, sr.top, sr.right - sr.left, sr.bottom - sr.top);
}
CV_INLINE RECT CvRectToRect(CvRect sr);
CV_INLINE RECT CvRectToRect(CvRect sr)
{
    RECT dr;
    dr.left = sr.x;
    dr.top = sr.y;
    dr.right = sr.x + sr.width;
    dr.bottom = sr.y + sr.height;

    return dr;
}
CV_INLINE IplROI RectToROI(RECT r);
CV_INLINE IplROI RectToROI(RECT r)
{
    IplROI roi;
    r = NormalizeRect(r);
    roi.xOffset = r.left;
    roi.yOffset = r.top;
    roi.width = r.right - r.left;
    roi.height = r.bottom - r.top;
    roi.coi = 0;

    return roi;
}
void  FillBitmapInfo(BITMAPINFO* bmi, int width, int height, int bpp, int origin)
{
    assert(bmi && width >= 0 && height >= 0 && (bpp == 8 || bpp == 24 || bpp == 32));

    BITMAPINFOHEADER* bmih = &(bmi->bmiHeader);

    memset(bmih, 0, sizeof(*bmih));
    bmih->biSize = sizeof(BITMAPINFOHEADER);
    bmih->biWidth = width;
    bmih->biHeight = origin ? abs(height) : -abs(height);
    bmih->biPlanes = 1;
    bmih->biBitCount = (unsigned short)bpp;
    bmih->biCompression = BI_RGB;
    if (bpp == 8)
    {
        RGBQUAD* palette = bmi->bmiColors;
        int i;
        for (i = 0; i < 256; i++)
        {
            palette[i].rgbBlue = palette[i].rgbGreen = palette[i].rgbRed = (BYTE)i;
            palette[i].rgbReserved = 0;
        }
    }
}
CvvImage::CvvImage()
{
    m_img = 0;
}
void CvvImage::Destroy()
{
    cvReleaseImage(&m_img);
}
CvvImage::~CvvImage()
{
    Destroy();
}
bool  CvvImage::Create(int w, int h, int bpp, int origin)
{
    const unsigned max_img_size = 10000;

    if ((bpp != 8 && bpp != 24 && bpp != 32) ||
        (unsigned)w >= max_img_size || (unsigned)h >= max_img_size ||
        (origin != IPL_ORIGIN_TL && origin != IPL_ORIGIN_BL))
    {
        assert(0); // most probably, it is a programming error
        return false;
    }
    if (!m_img || Bpp() != bpp || m_img->width != w || m_img->height != h)
    {
        if (m_img && m_img->nSize == sizeof(IplImage))
            Destroy();

        m_img = cvCreateImage(cvSize(w, h), IPL_DEPTH_8U, bpp / 8);
    }
    if (m_img)
        m_img->origin = origin == 0 ? IPL_ORIGIN_TL : IPL_ORIGIN_BL;
    return m_img != 0;
}
void  CvvImage::CopyOf(CvvImage& image, int desired_color)
{
    IplImage* img = image.GetImage();
    if (img)
    {
        CopyOf(img, desired_color);
    }
}
#define HG_IS_IMAGE(img)                                                  \
  ((img) != 0 && ((const IplImage*)(img))->nSize == sizeof(IplImage) && \
  ((IplImage*)img)->imageData != 0)
void  CvvImage::CopyOf(IplImage* img, int desired_color)
{
    if (HG_IS_IMAGE(img))
    {
        int color = desired_color;
        CvSize size = cvGetSize(img);
        if (color < 0)
            color = img->nChannels > 1;
        if (Create(size.width, size.height,
            (!color ? 1 : img->nChannels > 1 ? img->nChannels : 3) * 8,
            img->origin))
        {
            cvConvertImage(img, m_img, 0);
        }
    }
}
bool  CvvImage::Load(const char* filename, int desired_color)
{
    IplImage* img = cvLoadImage(filename, desired_color);
    if (!img)
        return false;

    CopyOf(img, desired_color);
    cvReleaseImage(&img);

    return true;
}
bool  CvvImage::LoadRect(const char* filename,
    int desired_color, CvRect r)
{
    if (r.width < 0 || r.height < 0) return false;

    IplImage* img = cvLoadImage(filename, desired_color);
    if (!img)
        return false;
    if (r.width == 0 || r.height == 0)
    {
        r.width = img->width;
        r.height = img->height;
        r.x = r.y = 0;
    }
    if (r.x > img->width || r.y > img->height ||
        r.x + r.width < 0 || r.y + r.height < 0)
    {
        cvReleaseImage(&img);
        return false;
    }

    if (r.x < 0)
    {
        r.width += r.x;
        r.x = 0;
    }
    if (r.y < 0)
    {
        r.height += r.y;
        r.y = 0;
    }
    if (r.x + r.width > img->width)
        r.width = img->width - r.x;

    if (r.y + r.height > img->height)
        r.height = img->height - r.y;
    cvSetImageROI(img, r);
    CopyOf(img, desired_color);
    cvReleaseImage(&img);
    return true;
}
bool  CvvImage::Save(const char* filename)
{
    if (!m_img)
        return false;
    cvSaveImage(filename, m_img);
    return true;
}
void  CvvImage::Show(const char* window)
{
    if (m_img)
        cvShowImage(window, m_img);
}
void  CvvImage::Show(HDC dc, int x, int y, int w, int h, int from_x, int from_y)
{
    if (m_img && m_img->depth == IPL_DEPTH_8U)
    {
        uchar buffer[sizeof(BITMAPINFOHEADER) + 1024];
        BITMAPINFO* bmi = (BITMAPINFO*)buffer;
        int bmp_w = m_img->width, bmp_h = m_img->height;
        FillBitmapInfo(bmi, bmp_w, bmp_h, Bpp(), m_img->origin);
        from_x = MIN(MAX(from_x, 0), bmp_w - 1);
        from_y = MIN(MAX(from_y, 0), bmp_h - 1);
        int sw = MAX(MIN(bmp_w - from_x, w), 0);
        int sh = MAX(MIN(bmp_h - from_y, h), 0);
        SetDIBitsToDevice(
            dc, x, y, sw, sh, from_x, from_y, from_y, sh,
            m_img->imageData + from_y * m_img->widthStep,
            bmi, DIB_RGB_COLORS);
    }
}
void  CvvImage::DrawToHDC(HDC hDCDst, RECT* pDstRect)
{
    if (pDstRect && m_img && m_img->depth == IPL_DEPTH_8U && m_img->imageData)
    {
        uchar buffer[sizeof(BITMAPINFOHEADER) + 1024];
        BITMAPINFO* bmi = (BITMAPINFO*)buffer;
        int bmp_w = m_img->width, bmp_h = m_img->height;
        CvRect roi = cvGetImageROI(m_img);
        CvRect dst = RectToCvRect(*pDstRect);
        if (roi.width == dst.width && roi.height == dst.height)
        {
            Show(hDCDst, dst.x, dst.y, dst.width, dst.height, roi.x, roi.y);
            return;
        }
        if (roi.width > dst.width)
        {
            SetStretchBltMode(
                hDCDst,          // handle to device context
                HALFTONE);
        }
        else
        {
            SetStretchBltMode(
                hDCDst,          // handle to device context
                COLORONCOLOR);
        }
        FillBitmapInfo(bmi, bmp_w, bmp_h, Bpp(), m_img->origin);
        ::StretchDIBits(
            hDCDst,
            dst.x, dst.y, dst.width, dst.height,
            roi.x, roi.y, roi.width, roi.height,
            m_img->imageData, bmi, DIB_RGB_COLORS, SRCCOPY);
    }
}
void  CvvImage::Fill(int color)
{
    cvSet(m_img, cvScalar(color & 255, (color >> 8) & 255, (color >> 16) & 255, (color >> 24) & 255));
}